package util

import (
	"fmt"
	"github.com/PuerkitoBio/goquery"
	"os"
	"path/filepath"
)

// 保存HTML到文件
func SaveHTMLToFile(doc *goquery.Document, dir, filename string) error {
	// 创建目录（如果不存在）
	if err := os.MkdirAll(dir, 0755); err != nil {
		return fmt.Errorf("error creating directory: %v", err)
	}

	// 创建文件路径
	filePath := filepath.Join(dir, filename)

	// 创建文件
	file, err := os.Create(filePath)
	if err != nil {
		return fmt.Errorf("error creating file: %v", err)
	}
	defer file.Close()

	// 将文档内容写入文件
	html, err := doc.Html()
	if err != nil {
		return fmt.Errorf("error getting HTML from document: %v", err)
	}

	_, err = file.WriteString(html)
	if err != nil {
		return fmt.Errorf("error writing to file: %v", err)
	}

	fmt.Printf("HTML content saved to %s\n", filePath)
	return nil
}
